﻿@echo off
setlocal EnableDelayedExpansion
goto :main

:activate_conda
    :: 设置 Miniconda3 安装路径和项目路径
    SET CONDA_PATH=D:\miniconda3
    SET PROJECT_PATH=E:\python\wemedia
    SET ENV_NAME=wemedia
    SET ENV_PATH=D:\venv\wemedia

    :: 显示欢迎信息
    echo ========================================
    echo    启动 %ENV_NAME% 虚拟环境中...
    echo    项目路径: %PROJECT_PATH%
    echo    环境路径: %ENV_PATH%
    echo ========================================
    echo.

    :: 尝试激活环境
    call %CONDA_PATH%\Scripts\activate.bat
    call conda activate %ENV_NAME%

    :: 如果失败，尝试直接激活环境路径
    if %ERRORLEVEL% NEQ 0 (
        echo 尝试直接激活环境路径...
        call %CONDA_PATH%\Scripts\activate.bat %ENV_PATH%
    )

    :: 检查是否成功
    if %ERRORLEVEL% NEQ 0 (
        echo 环境激活失败
        exit /b 1
    ) else (
        echo 环境激活成功
        exit /b 0
    )
goto :eof

:main
    :: 使用 UTF-8 编码
    chcp 65001 > nul
    :: 设置终端窗口标题
    TITLE Conda 虚拟环境启动器 - wemedia

    :: 切换到项目目录
    cd /d %PROJECT_PATH%

    :: 激活环境
    call :activate_conda
    if %ERRORLEVEL% NEQ 0 (
        echo.
        echo 环境激活失败，请检查环境设置是否正确。
        echo.
        echo 可用的conda环境:
        call conda env list
        pause
        exit /b 1
    )

    :: 显示Python版本和环境信息
    python --version
    echo.
    where python
    echo.

    :: 显示激活成功信息
    echo =======================================
    echo    已成功激活 %ENV_NAME% 环境
    echo    可以开始在该环境中工作了
    echo =======================================
    echo.

    :: 设置提示符
    prompt (%ENV_NAME%) $P$G
    
    :: 等待用户操作
    echo 现在可以在当前环境中工作了。按任意键继续...

    :: 启动新的命令行会话
    cd /d %PROJECT_PATH%
    cmd /k
goto :eof